Deep Learning

ONNXModel

Python

from synapse.ml.onnx import ONNXModel

model_path = "PUT_YOUR_MODEL_PATH"
onnx_ml = (ONNXModel()
            .setModelLocation(model_path)
            .setFeedDict({"float_input": "features"})
            .setFetchDict({"prediction": "output_label", "rawProbability": "output_probability"}))

Scala

import com.microsoft.azure.synapse.ml.onnx._

val model_path = "PUT_YOUR_MODEL_PATH"
val onnx_ml = (new ONNXModel()
                  .setModelLocation(model_path)
                  .setFeedDict(Map("float_input" -> "features"))
                  .setFetchDict(Map("prediction" -> "output_label", "rawProbability" -> "output_probability")))

Python API: ONNXModel

Scala API: ONNXModel

Source: ONNXModel